package net.bwie.realtime.jtp.dws.log.job;

import org.apache.flink.configuration.Configuration;
import org.apache.flink.table.api.EnvironmentSettings;
import org.apache.flink.table.api.Table;
import org.apache.flink.table.api.TableEnvironment;

/**
 * @Author: FuHe
 * @Date: 2025/5/23
 */
public class JtpTrafficAppStartMinuteWindowDwsJob {
    public static void main(String[] args) {
        // 表执行环境
        TableEnvironment tabEnv = getTableEnv();
        // 输入表  input 映射到kafka消息队列
        createInputTable(tabEnv);
        // 数据处理
        handle(tabEnv);
        // 输出表  output  映射到clickhouse表
        createOutputTable(tabEnv);
        // 保存数据
        saveToClickhouse(tabEnv);
        //todo关联广告维表数据
        joinDim(tabEnv);
    }

    private static void joinDim(TableEnvironment tabEnv) {
        tabEnv.executeSql("CREATE TABLE `ads_info_mysql_source_dim`\n" +
                "(\n" +
                "  id               INT COMMENT '主键ID',\n" +
                "  name             STRING COMMENT '广告名称',\n" +
                "  content          STRING COMMENT '广告内容（图片URL/视频URL/H5代码）',\n" +
                "  type             STRING COMMENT '广告类型（IMAGE, VIDEO, HTML5）',\n" +
                "  start_time       TIMESTAMP COMMENT '展示开始时间',\n" +
                "  end_time         TIMESTAMP COMMENT '展示结束时间',\n" +
                "  is_active        INT COMMENT '是否启用',\n" +
                "  target_condition STRING COMMENT '定向条件（JSON格式）',\n" +
                "  created_at       TIMESTAMP COMMENT '数据插入时间',\n" +
                "  updated_at       TIMESTAMP COMMENT '数据更新时间',\n" +
                "  PRIMARY KEY (id) NOT ENFORCED\n" +
                ") WITH (\n" +
                "  'connector' = 'jdbc',\n" +
                "  'url' = 'jdbc:mysql://node103:3306/jtp_ads',\n" +
                "  'table-name' = 'ads_info',\n" +
                "  'lookup.cache.max-rows' = '10',\n" +
                "  'lookup.cache.ttl' = '1 hour',\n" +
                "  'username' = 'root',\n" +
                "  'password' = '123456',\n" +
                "  'driver' = 'com.mysql.cj.jdbc.Driver'\n" +
                ")");
        Table table = tabEnv.sqlQuery("SELECT\n" +
                "    t1.window_start_time\n" +
                "    , t1.window_end_time\n" +
                "    , t2.name AS open_ad_name\n" +
                "    , t1.open_ad_count\n" +
                "    , t1.open_ad_ms\n" +
                "    , t1.open_ad_skip_ms\n" +
                "    , t1.ts\n" +
                "FROM start_ad_report_table t1\n" +
                "    LEFT JOIN ads_info_mysql_source_dim FOR SYSTEM_TIME AS OF t1.proc_time AS t2\n" +
                "        ON cast(t1.open_ad_id as int) = t2.id");
        table.execute().print();
    }

    private static void saveToClickhouse(TableEnvironment tabEnv) {
        tabEnv.executeSql("INSERT INTO dws_traffic_app_start_window_report_clickhouse_sink\n" +
                "SELECT\n" +
                "    DATE_FORMAT(window_start_time, 'yyyy-MM-dd HH:mm:ss') AS window_start_time\n" +
                "    , DATE_FORMAT(window_end_time, 'yyyy-MM-dd HH:mm:ss') AS window_end_time\n" +
                "    , entry\n" +
                "    , entry_count\n" +
                "    , loading_time\n" +
                "    , ts\n" +
                "FROM start_report_table");
//
        tabEnv.executeSql("INSERT INTO dws_traffic_app_start_ad_window_report_clickhouse_sink\n" +
                "SELECT\n" +
                "    DATE_FORMAT(window_start_time, 'yyyy-MM-dd HH:mm:ss') AS window_start_time\n" +
                "     , DATE_FORMAT(window_end_time, 'yyyy-MM-dd HH:mm:ss') AS window_end_time\n" +
                "     , open_ad_id\n" +
                "     , open_ad_count\n" +
                "     , open_ad_ms\n" +
                "     , open_ad_skip_ms\n" +
                "     , ts\n" +
                "FROM start_ad_report_table");
    }

    private static void createOutputTable(TableEnvironment tabEnv) {
        tabEnv.executeSql("CREATE TABLE dws_traffic_app_start_window_report_clickhouse_sink\n" +
                "(\n" +
                "    window_start_time STRING COMMENT '窗口开始日期时间',\n" +
                "    window_end_time   STRING COMMENT '窗口结束日期时间',\n" +
                "    entry             STRING COMMENT '启动app方式',\n" +
                "    entry_count       BIGINT COMMENT '启动app方式次数',\n" +
                "    loading_time      BIGINT COMMENT '启动app加载时间',\n" +
                "    ts                BIGINT COMMENT '数据计算时间戳'\n" +
                ") WITH (\n" +
                "    'connector' = 'clickhouse',\n" +
                "    'url' = 'jdbc:clickhouse://node103:8123/jtp_log_report',\n" +
                "    'table' = 'dws_traffic_app_start_window_report',\n" +
                "    'username' = 'default',\n" +
                "    'password' = '',\n" +
                "    'format' = 'json'\n" +
                ")");

        tabEnv.executeSql("CREATE TABLE dws_traffic_app_start_ad_window_report_clickhouse_sink\n" +
                "(\n" +
                "    `window_start_time` STRING COMMENT '窗口开始日期时间',\n" +
                "    `window_end_time`   STRING COMMENT '窗口结束日期时间',\n" +
                "    `open_ad_id`        STRING COMMENT '广告id',\n" +
                "    `open_ad_count`     BIGINT COMMENT '广告展示次数',\n" +
                "    `open_ad_ms`        BIGINT COMMENT '广告总共播放时间',\n" +
                "    `open_ad_skip_ms`   BIGINT COMMENT '用户跳过广告时间',\n" +
                "    `ts`                BIGINT COMMENT '数据计算时间戳'\n" +
                ") WITH (\n" +
                "    'connector' = 'clickhouse',\n" +
                "    'url' = 'jdbc:clickhouse://node103:8123/jtp_log_report',\n" +
                "    'table' = 'dws_traffic_app_start_ad_window_report',\n" +
                "    'username' = 'default',\n" +
                "    'password' = '',\n" +
                "    'format' = 'json'\n" +
                ")");
    }

    private static void handle(TableEnvironment tabEnv) {
        Table table2 = tabEnv.sqlQuery("SELECT\n" +
                "    `start`['entry'] AS entry\n" +
                "    , CAST(`start`['loading_time'] AS BIGINT) AS loading_time\n" +
                "    , `start`['open_ad_id'] AS open_ad_id\n" +
                "    , CAST(`start`['open_ad_ms'] AS BIGINT) AS open_ad_ms\n" +
                "    , CAST(`start`['open_ad_skip_ms'] AS BIGINT) AS open_ad_skip_ms\n" +
                "    , row_time\n" +
                "FROM dwd_log_start_log_kafka_source\n" +
                "WHERE `start`['entry'] IS NOT NULL");

        tabEnv.createTemporaryView("start_log_table", table2);

        Table table = tabEnv.sqlQuery("SELECT\n" +
                "    TUMBLE_START(row_time, INTERVAL '1' MINUTES) AS window_start_time\n" +
                "     , TUMBLE_END(row_time, INTERVAL '1' MINUTES) AS window_end_time\n" +
                "     , entry\n" +
                "     , count(entry) AS entry_count\n" +
                "     , sum(loading_time) AS loading_time\n" +
                "     , UNIX_TIMESTAMP() * 1000 AS ts\n" +
                "FROM start_log_table\n" +
                "GROUP BY\n" +
                "    TUMBLE(row_time, INTERVAL '1' MINUTES ),\n" +
                "    entry");
        tabEnv.createTemporaryView("start_report_table", table);

//        Table table1 = tabEnv.sqlQuery("SELECT\n" +
//                "    TUMBLE_START(row_time, INTERVAL '1' MINUTES) AS window_start_time\n" +
//                "     , TUMBLE_END(row_time, INTERVAL '1' MINUTES) AS window_end_time\n" +
//                "     , open_ad_id\n" +
//                "     , count(open_ad_id) AS open_ad_count\n" +
//                "     , sum(open_ad_ms) AS open_ad_ms\n" +
//                "     , sum(open_ad_skip_ms) AS open_ad_skip_ms\n" +
//                "     , UNIX_TIMESTAMP() * 1000 AS ts\n" +
//                "FROM start_log_table\n" +
//                "GROUP BY\n" +
//                "    TUMBLE(row_time, INTERVAL '1' MINUTES ),\n" +
//                "    open_ad_id");
        Table table1 = tabEnv.sqlQuery("SELECT\n" +
                "    window_start AS window_start_time,\n" +
                "    window_end AS  window_end_time,\n" +
                "    open_ad_id,\n" +
                "    count(open_ad_id) as open_ad_count,\n" +
                "    sum(open_ad_ms) as open_ad_ms,\n" +
                "    sum(open_ad_skip_ms) as open_ad_skip_ms,\n" +
                "    unix_timestamp()*1000 as ts,\n" +
                "    proctime() as proc_time\n" +
                "FROM table ( tumble(table start_log_table,descriptor(row_time),  INTERVAL '1' MINUTES )) t1\n" +
                "group by window_start,window_end,open_ad_id");

//        table1.execute().print();
        tabEnv.createTemporaryView("start_ad_report_table", table1);
    }

    private static void createInputTable(TableEnvironment tabEnv) {
        // 启动app日志
        tabEnv.executeSql("CREATE TABLE dwd_log_start_log_kafka_source (\n" +
                "    common MAP<STRING, STRING> COMMENT '公共环境信息',\n" +
                "    `start` MAP<STRING, STRING> COMMENT '启动信息',\n" +
                "    ts BIGINT COMMENT '时间戳',\n" +
                "    row_time AS TO_TIMESTAMP(FROM_UNIXTIME(ts / 1000, 'yyyy-MM-dd HH:mm:ss.SSS')) COMMENT '事件时间，指定格式',\n" +
                "    WATERMARK FOR row_time AS row_time - INTERVAL '0' MINUTE\n" +
                ") WITH (\n" +
                "    'connector' = 'kafka',\n" +
                "    'topic' = 'dwd-traffic-start-log',\n" +
                "    'properties.bootstrap.servers' = 'node101:9092,node102:9092,node103:9092',\n" +
                "    'properties.group.id' = 'gid_dwd_traffic_start_log',\n" +
                "    'scan.startup.mode' = 'earliest-offset',\n" +
                "    'format' = 'json',\n" +
                "    'json.fail-on-missing-field' = 'false',\n" +
                "    'json.ignore-parse-errors' = 'true'\n" +
                ")");
    }

    private static TableEnvironment getTableEnv() {
        // 环境属性设置
        EnvironmentSettings settings = EnvironmentSettings.newInstance()
                .inStreamingMode()
                .useBlinkPlanner()
                .build();
        // 表执行环境
        TableEnvironment tabEnv = TableEnvironment.create(settings);
        //  配置属性设置
        Configuration configuration = tabEnv.getConfig().getConfiguration();
        // 时区设置
        configuration.setString("table.local-time-zone", "Asia/Shanghai");
        // 并发设置
        configuration.setString("table.exec.resource.default-parallelism", "1");
        // 状态保存时间
        configuration.setString("table.exec.state.ttl", "5s");
        // 返回值
        return tabEnv;
    }
}
